from functools import wraps


def sqlalchemy_context():
    def add_context(func):
        @wraps(func)
        def do_job(*args, **kwargs):
            # app.app_context().push()
            result = func(*args, **kwargs)
            return result

        return do_job

    return add_context
